#pragma once
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include "Log.hpp"
#include "Thread.hpp"
#include "Cond.hpp"
#include "Mutex.hpp"

namespace ThreadPoolModule
{
    using namespace ThreadModlue;
    using namespace LogModule;
    using namespace CondModule;
    using namespace MutexModule;

    static const int gnum = 5;
    template <typename T>
    class ThreadPool
    {
    private:
        void WakeUpAllThread()
        {
            LockGuard lockguard(_mutex);
            if (_sleepnum)
                _cond.Broadcast();

            LOG(LogLevel::INFO) << "唤醒所有的休眠线程";
        }

        void WakeUpOne()
        {
            //不能加锁 
            //LockGuard lockguard(_mutex);
            //pthread_cond_signal 函数的作用是发送一个信号给另外一个正在处于阻塞等待状态的线程,
            // 使其脱离阻塞状态,继续执行.如果没有线程处在阻塞等待状态,pthread_cond_signal也会成功返回。
            _cond.Signal();

            LOG(LogLevel::INFO) << "唤醒一个休眠线程";
        }   

        ThreadPool(int num = gnum)
        : _num(num),
        _sleepnum(0),
        _isrunning(false)
    {
        for (int i = 0; i < num; i++)
        {
            _threads.emplace_back(
                [this]()
                {
                    HandlerTask();
                });
        }
    }

    void Start()
    {
        if (_isrunning)
            return;
        _isrunning = true;
        for (auto &thread : _threads)
        {
            thread.Start();
            LOG(LogLevel::INFO) << "start new thread success: " << thread.Name();
        }
    }

        //默认 线程池是不能被拷贝和 赋值的
        ThreadPool(const ThreadPool<T>& ) = delete;
        ThreadPool<T> operator=(const ThreadPool<T>& ) = delete;

    public:
        ///?????????   为什么要返回 threadpool<T>
        
         static ThreadPool<T>* GetInstance()
        {
            if(inc == nullptr)
            {

                //可能多个线程来拿锁， 所以要多家一遍判断 以防inc 被获取多遍案例
                LockGuard lockguard(_lock);

                LOG(LogLevel::DEBUG) << "获取单例....";
                if(inc == nullptr)
                {
                    LOG(LogLevel::DEBUG) << "首次获取案例, 创建之....";
                    inc = new ThreadPool<T>();
                    inc->Start();
                }
            }
            return inc;
        }


        void Stop()
        {
            //std::cout << _isrunning << std::endl;
            if (!_isrunning)
                return;
            _isrunning = false;

            WakeUpAllThread();
            //LOG(LogLevel::INFO) << "线程池 stop ";
        }

        void Join()
        {
            for (auto thread : _threads)
            {
                thread.Join();
            }
        }

        void HandlerTask()
        {
            char name[128];
            pthread_getname_np(pthread_self(), name, sizeof(name));

            while (true)
            {
                T t;
                {
                    LockGuard lockguard(_mutex);
                    //处理任务需要时间
                    //1. a.队列为空 b. 线程池没有退出
                    //b 线程池若退出了 且 还剩一个任务 一个被唤醒的线程 死循环，无法吧最后一任务给解决了；
                    // 线程池关机后 优先 执行任务
                    while (_taskq.empty() && _isrunning)
                    {
                        _sleepnum++;
                        _cond.Wait(_mutex);
                        _sleepnum--;
                    }

                    if(_taskq.empty() && !_isrunning)
                    {
                        LOG(LogLevel::INFO) << name << " 退出了, 线程池退出&&任务队列为空";
                        break;
                    }

                    // 一定有任务
                    t = _taskq.front(); // 从q中获取任务，任务已经是线程私有的了！！！
                    _taskq.pop();
                }
                 t(); // 处理任务，需要在临界区内部处理吗？1 0
            }
        }

        bool Enqueue(const T &in)
        {
            if (_isrunning)
            {
                LockGuard lockguard(_mutex);
                
                //先给任务 在唤醒线程
                _taskq.push(in);
                if (_threads.size() == _sleepnum)
                    WakeUpOne();
                //LOG(LogLevel::INFO) << "获取任务 成功";
                return true;  
            }
            return false;
        }
        ~ThreadPool()
        {
        }

    private:
        std::vector<Thread> _threads;
        int _num; // 线程池中，线程的个数

        bool _isrunning; // 线程数池开关
        int _sleepnum;

        std::queue<T> _taskq;
        Cond _cond;
        Mutex _mutex;

        //bug?? -> 需要在定义一个 单例使用的锁 毕竟获取单例 是 静态函数
        static ThreadPool<T>* inc;
        static Mutex _lock;
    };

    template <typename T>
    ThreadPool<T>* ThreadPool<T>::inc = nullptr;

    template <typename T>
    Mutex ThreadPool<T>::_lock;
}